package demo;

import java.util.HashMap;
import java.util.Map;

/**
 * @Author liangzai
 * @Description:
 */
public class Training {
    //统计坏数对的数目
    public long countBadPairs(int[] nums) {
        Map<Long, Integer> map = new HashMap<>();
        long res = 0;
        for (int i = 0; i < nums.length; i++) {
            long p = i - nums[i];
            res += (i - map.getOrDefault(p, 0));
            map.put(p, map.getOrDefault(p, 0) + 1);
        }
        return res;
    }
}
